package pers.qianyu.month_202012.date_20201206;

import java.util.*;

/**
 * [同 date_20201123]
 * 416. 分割等和子集
 * https://leetcode-cn.com/problems/partition-equal-subset-sum/
 *
 * @author mizzle rain
 * @date 2020-12-06 14:27
 */
public class CanPartition {
    public boolean canPartition(int[] nums) {
        int sum = Arrays.stream(nums).sum();
        if ((sum & 1) == 1) {
            return false;
        }
        sum >>= 1;
        // dp 含义：数组的前 i 个数 往 容量为 j 的背包里装，背包里装的最大质量
        int[][] dp = new int[nums.length + 1][sum + 1];
        for (int i = 1; i < dp.length; i++) {
            for (int k = 1; k < dp[i].length; k++) {
                if (k - nums[i - 1] < 0) {
                    dp[i][k] = dp[i - 1][k];
                } else {
                    dp[i][k] = Math.max(dp[i - 1][k],
                            dp[i - 1][k - nums[i - 1]] + nums[i - 1]);
                }
            }
        }
        // 容量为 sum 的背包正好装满
        return dp[nums.length][sum] == sum;
    }
}
